/*
 * Copyright (c) 2012 Robert Burrell Donkin robertburrelldonkin.name
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy of this software 
 * and associated documentation files (the "Software"), to deal in the Software without restriction, 
 * including without limitation the rights to use, copy, modify, merge, publish, distribute, 
 * sublicense, and/or sell copies of the Software, and to permit persons to whom the Software 
 * is furnished to do so, subject to the following conditions:
 * 
 * The above copyright notice and this permission notice shall be included in all copies or 
 * substantial portions of the Software.
 * 
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING 
 * BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND 
 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, 
 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 * 
 */
package walrus.dir;

import java.io.File;
import java.io.FileFilter;
import java.util.concurrent.BlockingQueue;

import name.robertburrelldonkin.alfie.Message;
import name.robertburrelldonkin.alfie.QueuingSource;

import org.apache.commons.io.filefilter.DirectoryFileFilter;
import org.apache.commons.io.filefilter.TrueFileFilter;



public class DirectorySource extends QueuingSource<File> {

	private FileFilter filter = TrueFileFilter.TRUE;
	private File file;

	public DirectorySource(final BlockingQueue<Message<File>> out) {
		super(out);
	}
	
	public FileFilter getFilter() {
		return filter;
	}

	public DirectorySource setFilter(FileFilter filter) {
		this.filter = filter;
		return this;
	}

	public void walk(final File directory) {
		if (directory.isDirectory()) {
			for (final File file: directory.listFiles(filter)) {
				output(file);
			}
			for (final File file: directory.listFiles((FileFilter) DirectoryFileFilter.INSTANCE)) {
				walk(file);
			}
		} else {
			output(directory);
		}
	}

	public File getFile() {
		return file;
	}

	public DirectorySource setFile(File file) {
		this.file = file;
		return this;
	}

	public void open() {
		walk(file);
	}
}
